import torch
import torch.nn as nn

# 随机初始化词向量矩阵：这种方式很容易理解，就是使用self.embedding = torch.nn.Embedding(vocab_size, embed_dim)命令直接随机生成个初始化的词向量矩阵，
# 此时的向量值符合正态分布N(0,1)，这里的vocab_size是指词向量矩阵能表征的词的个数
embedding = nn.Embedding(10, 3)  # an Embedding module containing 10 tensors of size 3  10个张量，每个张量的维度为3
input = torch.LongTensor([[1, 2, 4, 5], [4, 3, 2, 9]])  # a batch of 2 samples of 4 indices each  两个样本，每个样本有四个索引
e = embedding(input)
print(e)